/*
 * Copyright (C) 2023 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
import hilog from "@ohos.hilog";
import TestRunner from "@ohos.application.testRunner";
import AbilityDelegatorRegistry from "@ohos.app.ability.abilityDelegatorRegistry";

var abilityDelegator = undefined;
var abilityDelegatorArguments = undefined;

async function onAbilityCreateCallback() {
  hilog.info(0x0000, "testTag", "%{public}s", "onAbilityCreateCallback");
}

async function addAbilityMonitorCallback(err: any) {
  hilog.info(
    0x0000,
    "testTag",
    "addAbilityMonitorCallback : %{public}s",
    JSON.stringify(err) ?? ""
  );
}

export default class OpenHarmonyTestRunner implements TestRunner {
  constructor() {}

  onPrepare() {
    hilog.info(
      0x0000,
      "testTag",
      "%{public}s",
      "OpenHarmonyTestRunner OnPrepare "
    );
  }

  async onRun() {
    hilog.info(
      0x0000,
      "testTag",
      "%{public}s",
      "OpenHarmonyTestRunner onRun run"
    );
    abilityDelegatorArguments = AbilityDelegatorRegistry.getArguments();
    abilityDelegator = AbilityDelegatorRegistry.getAbilityDelegator();
    var testAbilityName = abilityDelegatorArguments.bundleName + ".TestAbility";
    let lMonitor = {
      abilityName: testAbilityName,
      onAbilityCreate: onAbilityCreateCallback,
    };
    abilityDelegator.addAbilityMonitor(lMonitor, addAbilityMonitorCallback);
    var cmd =
      "aa start -d 0 -a TestAbility" +
      " -b " +
      abilityDelegatorArguments.bundleName;
    var debug = abilityDelegatorArguments.parameters["-D"];
    if (debug == "true") {
      cmd += " -D";
    }
    hilog.info(0x0000, "testTag", "cmd : %{public}s", cmd);
    abilityDelegator.executeShellCommand(cmd, (err: any, d: any) => {
      hilog.info(
        0x0000,
        "testTag",
        "executeShellCommand : err : %{public}s",
        JSON.stringify(err) ?? ""
      );
      hilog.info(
        0x0000,
        "testTag",
        "executeShellCommand : data : %{public}s",
        d.stdResult ?? ""
      );
      hilog.info(
        0x0000,
        "testTag",
        "executeShellCommand : data : %{public}s",
        d.exitCode ?? ""
      );
    });
    hilog.info(
      0x0000,
      "testTag",
      "%{public}s",
      "OpenHarmonyTestRunner onRun end"
    );
  }
}
